import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;

/**
 * 如果字符串中的所有字符都相同，那么这个字符串是单字符重复的字符串。
 * 给你一个字符串text，你只能交换其中两个字符一次或者什么都不做，然后得到一些单字符重复的子串。返回其中最长的子串的长度。
 * <p>
 * 示例 1：
 * 输入：text = "ababa"
 * 输出：3
 * <p>
 * 示例 2：
 * 输入：text = "aaabaaa"
 * 输出：6
 * <p>
 * 示例 3：
 * 输入：text = "aaabbaaa"
 * 输出：4
 * <p>
 * 示例 4：
 * 输入：text = "aaaaa"
 * 输出：5
 * <p>
 * 示例 5：
 * 输入：text = "abcdef"
 * 输出：1
 * <p>
 * 提示：
 * 1 <= text.length <= 20000
 * text 仅由小写英文字母组成。
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/swap-for-longest-repeated-character-substring
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 */
public class Q01156m {

    public int maxRepOpt1(String text) {
        int ans = 0;
        Set<Integer>[] cs = new Set[123];
        int[] cm = new int[123];
        for (int i = 0; i < text.length(); i++) {
            char c = text.charAt(i);
            if (cs[c] == null) {
                cs[c] = new HashSet<>();
            }
            cs[c].add(i);
            cm[c] = i;
        }
        Map<Integer, Integer> lengthMap = new HashMap<>();
        for (int i = 'a'; i < cs.length; i++) {
            lengthMap.clear();
            Set<Integer> ls = cs[i];
            if (ls != null) {
                for (Integer j : ls) {
                    if (!ls.contains(j - 1)) {
                        int k = j;
                        while (ls.contains(k + 1)) {
                            k++;
                        }
                        int distance = k - j + 1;
                        Integer lastLength = lengthMap.getOrDefault(j - 2, 0);
                        ans = Math.max(ans, distance // 当前连续长度
                                + (lastLength > 0 ? (lengthMap.size() > 1 || k < cm[i]) ? lastLength + 1 : lastLength // 可以与前一个连上
                                : (lengthMap.size() > 0 || k < cm[i]) ? 1 : 0) // 连不上，但有落单的一个
                        );
                        lengthMap.put(k, distance);
                    }
                }
            }
        }
        return ans;
    }
}
